#!/bin/bash

stage=1

#attention=luong # scaled_luong | bahdanau | normed_bahdanau or ""
attention=bahdanau 
nmt_data=./data/wmt16_de_en
nmt_model=./nmt_model/wmt16_de_en/$attention


rm -rf $nmt_model

if [ $stage -eq 0 ]; then
	./scripts/wmt16_en_de.sh $nmt_data
fi

if [ $stage -eq 1 ]; then
	CUDA_VISIBLE_DEVICES=0,1 python -m nmt.nmt \
		--src=de --tgt=en \
		--hparams_path=nmt/standard_hparams/wmt16_gnmt_4_layer.json \
		--out_dir=$nmt_model \
		--vocab_prefix=$nmt_data/vocab.bpe.32000 \
		--train_prefix=$nmt_data/train.tok.clean.bpe.32000 \
		--dev_prefix=$nmt_data/newstest2013.tok.bpe.32000 \
		--test_prefix=$nmt_data/newstest2015.tok.bpe.32000 \
		--num_gpus 2
fi
